import os
import search_multi_all
#import shutil

def mkdir_output(output_dir):
    existence = os.path.exists(output_dir)
    if not existence:
        print('创建存放目录')
        os.makedirs(output_dir)    # 创建目录
        os.chdir(output_dir)       # 切换到创建的文件夹
        return True
    else:
        print('目录已存在,即将保存！')
        return False


if __name__ == '__main__':
    filepath = "/home/wind/code/db-export/"   # 待转换文件存放的路径
    os.chdir(filepath)                  # 切换到改路径下
    filename = os.listdir(filepath)     # 得到文件夹下的所有文件名称

    #output_dir = '/home/wind/docs/'    # 转换后文件存放的路径
    #mkdir_output(output_dir)
    for i in range(len(filename)):
        if "txt" in filename[i]:
            changefile = filename[i]
            outputfile = filename[i].replace('txt', 'xlsx')
            #shutil.copyfile(changefile,outputfile)
            search_multi_all.convert(filepath,changefile,outputfile)
        
    print('全部转换完成')   